#

class ClipConfig(object):
    def __init__(self):
        self.name = 'apps.clip.clip_config.ClipConfig'

    emb_dim = 192 # 图像和文本向量维度，缺省值32
    vit_width = 9
    img_size = (28,28)
    patch_size = (14,14)
    n_channels = 1
    vit_layers = 3
    vit_heads = 3
    vocab_size = 256
    text_width = 192 # 文本维度，缺省值32
    max_seq_length = 196 # 序列长度
    text_heads = 8
    text_layers = 4
    lr = 1e-3
    epochs = 10
    batch_size = 128
    ckpt_fn = './work/ckpt/clip.pt'
    device = 'cpu'